"""
1791. 找出星型图的中心节点
有一个无向的 星型 图，由 n 个编号从 1 到 n 的节点组成。星型图有一个 中心 节点，并且恰有 n - 1 条边将中心节点与其他每个节点连接起来。
给你一个二维整数数组 edges ，其中 edges[i] = [ui, vi] 表示在节点 ui 和 vi 之间存在一条边。请你找出并返回 edges 所表示星型图的中心节点。
示例 1：
        4
        |
        2
      /   \
    1       3
输入：edges = [[1,2],[2,3],[4,2]]输出：2
解释：如上图所示，节点 2 与其他每个节点都相连，所以节点 2 是中心节点。
示例 2：输入：edges = [[1,2],[5,1],[1,3],[1,4]] 输出：1
提示：

3 <= n <= 105
edges.length == n - 1
edges[i].length == 2
1 <= ui, vi <= n
ui != vi
题目数据给出的 edges 表示一个有效的星型图
"""
from mode import List


class Solution:
    def findCenter(self, edges: List[List[int]]) -> int:
        N = len(edges)
        count = [0] * (N + 2)
        for i, j in edges:
            count[i] += 1
            count[j] += 1
        for i in range(1, N + 2):
            if count[i] == N:
                return i
        return -1


class Solution1:
    def findCenter(self, edges: List[List[int]]) -> int:
        n = len(edges)
        count = [0] * (n + 2)
        for i, j in edges:
            count[i] += 1
            count[j] += 1
        for i in range(n+2):
            if count[i] == n:
                return i
        return -1


if __name__ == "__main__":
    # edges = [[1, 2], [2, 3], [4, 2]]
    edges = [[1, 18], [18, 2], [3, 18], [18, 4], [18, 5], [6, 18], [18, 7], [18, 8], [18, 9], [18, 10], [18, 11],
             [12, 18], [18, 13], [18, 14], [15, 18], [16, 18], [17, 18]]
    A = Solution1()
    print(A.findCenter(edges))
